from django import template

from ShoppingMall.models import Cake

register = template.Library()  # 以上两行代码是获取到Django模板所有tags和filter的library，以便我们写入一个新的方法


def cakename_by_id(id):  # 自定义方法，adict是我们使用这个方法的调用者，key是dict_get方法的参数，如果不太懂，接着往下看
    """get the value using key from a dict"""
    cake = Cake.objects.filter(id=id)
    return cake.name

def cakeprice_by_id(id):  # 自定义方法，adict是我们使用这个方法的调用者，key是dict_get方法的参数，如果不太懂，接着往下看
    """get the value using key from a dict"""
    cake = Cake.objects.filter(id=id)
    return cake.price

register.filter('cakename_by_id', cakename_by_id)  # 将此方法添加到模板中
register.filter('cakeprice_by_id', cakeprice_by_id)  # 将此方法添加到模板中
